
from api.ihrm_login_api import IhrmLogin


def get_headers():
    req_body = {"mobile": "13800000002", "password": "123456"}
    resp = IhrmLogin.login(req_body)

    token = resp.json().get("data")

    req_headers = {"Authorization": token}

    return req_headers


if __name__ == "__main__":
    print(get_headers())